package com.muyu.util;

import javax.servlet.http.HttpServletRequest;
import java.util.Arrays;
import java.util.List;

/**
 * @ClassName: IpUtils
 * @Author: Zirun.Xin
 * @Date: 2025/3/20 下午4:39
 * @Version: 1.0
 * @Description: IP地址工具
 */
public class IpUtils {

    /**
     * 获取客户端的真实 IP 地址
     *
     * @param request HTTP 请求对象
     * @return 客户端的真实 IP 地址
     */
    public static String getIpAddr(HttpServletRequest request) {
        if (request == null) {
            return "unknown";
        }

        // 1. 尝试从请求头中获取 IP 地址
        String ip = request.getHeader("X-Forwarded-For");
        if (isUnknown(ip)) {
            ip = request.getHeader("Proxy-Client-IP");
        }
        if (isUnknown(ip)) {
            ip = request.getHeader("WL-Proxy-Client-IP");
        }
        if (isUnknown(ip)) {
            ip = request.getHeader("HTTP_CLIENT_IP");
        }
        if (isUnknown(ip)) {
            ip = request.getHeader("HTTP_X_FORWARDED_FOR");
        }

        // 2. 如果从请求头中获取不到 IP 地址，则使用 request.getRemoteAddr()
        if (isUnknown(ip)) {
            ip = request.getRemoteAddr();
        }

        // 3. 处理多级代理的情况（例如 X-Forwarded-For: client, proxy1, proxy2）
        if (ip != null && ip.contains(",")) {
            ip = ip.split(",")[0].trim(); // 取第一个 IP 地址
        }

        return ip;
    }

    /**
     * 判断 IP 地址是否未知
     *
     * @param ip IP 地址
     * @return 是否未知
     */
    private static boolean isUnknown(String ip) {
        return ip == null || ip.length() == 0 || "unknown".equalsIgnoreCase(ip);
    }

    /**
     * 判断 IP 地址是否为本地地址或内网地址
     *
     * @param ip IP 地址
     * @return 是否为本地地址或内网地址
     */
    public static boolean isLocalOrInternal(String ip) {
        if (ip == null || ip.length() == 0) {
            return false;
        }

        // 本地地址
        if ("127.0.0.1".equals(ip)) {
            return true;
        }

        // IPv6 本地地址
        if ("0:0:0:0:0:0:0:1".equals(ip)) {
            return true;
        }

        // 内网地址范围
        List<String> internalIpPrefixes = Arrays.asList("10.", "172.", "192.168.");
        for (String prefix : internalIpPrefixes) {
            if (ip.startsWith(prefix)) {
                return true;
            }
        }

        return false;
    }
}
